cd ../..

gpus=$1


for exp in 5
do 
    python domain_agnostic_main.py \
        --gpus ${gpus}, \
        --opt_conf ./configs/opt_domain_agnostic.yaml \
        --exp_name da_exp_loss/TEST \
        --feat_dim 256 \
        --batch_sizes "[0, 12, 12, 96]" \
        --grad_compensate False \
        --joint_itpl_exponent ${exp} \
        --joint_itpl_warmup 50 \
        --n_epochs 200 \
        --test_from_checkpoint "/home/shaonian/SED/sssl_sed/codes/exp/da_exp_loss/baseline_reproduce/version_0/ckpts/epoch=289-obj_metric=1.457.ckpt"
done